/*
 * @BEGIN LICENSE
 *
 * Psi4: an open-source quantum chemistry software package
 *
 * Copyright (c) 2007-2024 The Psi4 Developers.
 *
 * The copyrights for code used from other parties are included in
 * the corresponding files.
 *
 * This file is part of Psi4.
 *
 * Psi4 is free software; you can redistribute it and/or modify
 * it under the terms of the GNU Lesser General Public License as published by
 * the Free Software Foundation, version 3.
 *
 * Psi4 is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public License along
 * with Psi4; if not, write to the Free Software Foundation, Inc.,
 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
 *
 * @END LICENSE
 */

#ifndef BLAS_H
#define BLAS_H

/**
 * fortran-ordered blas routines
 */

#include "psi4/pragma.h"
#include "blas_mangle.h"

typedef long int integer;
typedef double doublereal;

namespace psi {
namespace fnocc {

/**
 * fortran-ordered dgemv
 */
void PSI_API F_DGEMV(char trans, integer m, integer n, doublereal alpha, doublereal *A, integer lda, doublereal *X,
                     integer incx, doublereal beta, doublereal *Y, integer incy);
/**
 * fortran-ordered dgemm
 */
void PSI_API F_DGEMM(char transa, char transb, integer m, integer n, integer k, doublereal alpha, doublereal *A,
                     integer lda, doublereal *B, integer ldb, doublereal beta, doublereal *C, integer ldc);

/**
 * name mangling for fortran-ordered dgemv
 */
extern "C" {
void dgemv(char &trans, integer &m, integer &n, doublereal &alpha, doublereal *A, integer &lda, doublereal *X,
           integer &incx, doublereal &beta, doublereal *Y, integer &incy);
}
inline void DGEMV(char &trans, integer &m, integer &n, doublereal &alpha, doublereal *A, integer &lda, doublereal *X,
                  integer &incx, doublereal &beta, doublereal *Y, integer &incy) {
    dgemv(trans, m, n, alpha, A, lda, X, incx, beta, Y, incy);
}
/**
 * name mangling for fortran-ordered dgemm
 */
extern "C" {
void dgemm(char &transa, char &transb, integer &m, integer &n, integer &k, doublereal &alpha, doublereal *A,
           integer &lda, doublereal *B, integer &ldb, doublereal &beta, doublereal *C, integer &ldc);
}
inline void DGEMM(char &transa, char &transb, integer &m, integer &n, integer &k, doublereal &alpha, doublereal *A,
                  integer &lda, doublereal *B, integer &ldb, doublereal &beta, doublereal *C, integer &ldc) {
    dgemm(transa, transb, m, n, k, alpha, A, lda, B, ldb, beta, C, ldc);
}
/**
 * name mangling dcopy
 */
extern "C" {
void dcopy(integer &n, doublereal *dx, integer &incx, doublereal *dy, integer &incy);
}
inline void DCOPY(integer &n, doublereal *dx, integer &incx, doublereal *dy, integer &incy) {
    dcopy(n, dx, incx, dy, incy);
}
/**
 * name mangling dnrm2
 */
extern "C" {
double dnrm2(integer &N, doublereal *X, integer &INCX);
}
inline double DNRM2(integer &N, doublereal *X, integer &INCX) { return dnrm2(N, X, INCX); }
/**
 * name mangling dgesv
 */
extern "C" {
void dgesv(integer &N, integer &NRHS, doublereal *A, integer &LDA, integer *IPIV, doublereal *B, integer &LDB,
           integer &INFO);
}
inline void DGESV(integer &N, integer &NRHS, doublereal *A, integer &LDA, integer *IPIV, doublereal *B, integer &LDB,
                  integer &INFO) {
    dgesv(N, NRHS, A, LDA, IPIV, B, LDB, INFO);
}
/**
 * name mangling ddot
 */
extern "C" {
double ddot(integer &n, doublereal *dx, integer &incx, doublereal *dy, integer &incy);
}
inline double DDOT(integer &n, doublereal *dx, integer &incx, doublereal *dy, integer &incy) {
    return ddot(n, dx, incx, dy, incy);
}

/**
 * diagonalize a real symmetric matrix
 */
void Diagonalize(integer N, doublereal *A, doublereal *W);
/**
 * name mangling dsyev
 */
extern "C" {
void dsyev(char &JOBZ, char &UPLO, integer &N, doublereal *A, integer &LDA, doublereal *W, doublereal *WORK,
           integer &LWORK, integer &INFO);
}
inline void DSYEV(char &JOBZ, char &UPLO, integer &N, doublereal *A, integer &LDA, doublereal *W, doublereal *WORK,
                  integer &LWORK, integer &INFO) {
    dsyev(JOBZ, UPLO, N, A, LDA, W, WORK, LWORK, INFO);
}
/**
 * diagonalize a real symmetric packed matrix
 */
void Diagonalize2(integer N, doublereal *AP, doublereal *W, doublereal *Z);
/**
 * name mangling dspev
 */
extern "C" {
void dspev(char &JOBZ, char &UPLO, integer &N, doublereal *AP, doublereal *W, doublereal *Z, integer &LDZ,
           doublereal *WORK, integer &INFO);
}
inline void DSPEV(char &JOBZ, char &UPLO, integer &N, doublereal *AP, doublereal *W, doublereal *Z, integer &LDZ,
                  doublereal *WORK, integer &INFO) {
    dspev(JOBZ, UPLO, N, AP, W, Z, LDZ, WORK, INFO);
}

/**
 *  General SVD
 */
void SVD(integer M, integer N, doublereal *A, doublereal *U, doublereal *VT, doublereal *S);
/**
 * name mangling dgesvd
 */
extern "C" {
void dgesvd(char &JOBU, char &JOBVT, integer &M, integer &N, doublereal *A, integer &LDA, doublereal *S, doublereal *U,
            integer &LDU, doublereal *VT, integer &LDVT, doublereal *WORK, integer &LWORK, integer &INFO);
}
inline void DGESVD(char &JOBU, char &JOBVT, integer &M, integer &N, doublereal *A, integer &LDA, doublereal *S,
                   doublereal *U, integer &LDU, doublereal *VT, integer &LDVT, doublereal *WORK, integer &LWORK,
                   integer &INFO) {
    dgesvd(JOBU, JOBVT, M, N, A, LDA, S, U, LDU, VT, LDVT, WORK, LWORK, INFO);
}
}
}

#endif
